package com.qd.panda.service.space;

import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.DateUtil;
import com.cdqidi.util.JacksonUtil;
import com.cdqidi.util.RedisTemplateUtil;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.qd.common.panda.constant.PandaConstant;
import com.qd.common.panda.domain.entity.space.ParkSpaceDTO;
import com.qd.common.panda.domain.entity.space.ParkSpaceDateDTO;
import com.qd.common.panda.domain.entity.space.ParkSpaceTenantDTO;
import com.qd.panda.mapper.space.ParkSpaceDateMapper;
import com.qd.panda.model.ParkSpaceDate;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.time.LocalDateTime;
import java.util.*;

import static java.util.stream.Collectors.toList;
import static java.util.stream.Collectors.toSet;

/**
 * 租赁日期列表
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class ParkSpaceDateService extends BaseService<ParkSpaceDateMapper, ParkSpaceDate> {

    private static final String PREFIX_ID = "panda_park_space_date_did_{0}";
    private static final String PREFIX_LIST = "panda_park_space_date_sid_{0}";
    private static final String PREFIX_UNIQUE_PANDA_PARK = "panda_park_space_date_tenant_{0}_leaseTime_{1}_sid_{2}";

    private final RedisTemplateUtil<ParkSpaceDate> redisTemplateUtil;
    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    public ParkSpaceDateDTO save(ParkSpaceDateDTO dto) {
        checkUnique(dto);
        final ParkSpaceDate model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public ParkSpaceDateDTO update(ParkSpaceDateDTO dto) {
        checkUnique(dto);
        final ParkSpaceDate historyModel = Optional.ofNullable(getById(dto.getDid())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final ParkSpaceDate model = dtoToModel(dto);
        try{
            if (super.updateById(model)) {
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        }finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 批量添加
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void saveBatch(List<ParkSpaceDate> mList) {
        try {
            if (!super.saveBatch(mList)) {
                throw new ApiException("批量添加失败");
            }
        } finally {
            clearCaches(mList);
        }
    }

    /**
     * 批量更新
     *
     * @param mList 数据库对象
     */
    public void updateBatchIds(List<ParkSpaceDate> mList) {
        final List<String> ids = mList.stream().map(ParkSpaceDate::getDid).collect(toList());
        final List<ParkSpaceDate> historyModelList = checkIds(ids);
        try {
            if (!super.updateBatchById(mList)) {
                throw new ApiException("批量更新失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
            ids.clear();
        }
    }

    /**
     * 删除，单个
     *
     * @param did 主键 
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String did) {
        final ParkSpaceDate historyModel = Optional.ofNullable(getById(did)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(did)) {
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
        }
    }

    /**
     * 批量删除
     *
     * @param ids 主键集合
     */
    public void removeBatchIds(List<String> ids) {
        final List<ParkSpaceDate> historyModelList = checkIds(ids);
        try {
            if (!super.removeByIds(ids)) {
                throw new ApiException("批量删除失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
        }
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(ParkSpaceDateDTO dto) {
        final QueryWrapper<ParkSpaceDate> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param did 主键 
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String did) {
        boolean exists = false;
        if (StringUtils.hasLength(did)) {
            final ParkSpaceDate model = getById(did);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param did 主键 
     * @return 单条记录
     */
    public ParkSpaceDate getById(String did) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, did);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(did), ParkSpaceDate.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param did 主键 
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public ParkSpaceDateDTO getByIdDto(String did) {
        return Optional.ofNullable(getById(did)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    public ParkSpaceDate getOne(ParkSpaceDateDTO dto) {
        final QueryWrapper<ParkSpaceDate> queryWrapper = createQueryWrapper(dto);
        ParkSpaceDate model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public ParkSpaceDateDTO getOneDto(ParkSpaceDateDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    public List<ParkSpaceDate> getList(ParkSpaceDateDTO dto) {
        final QueryWrapper<ParkSpaceDate> queryWrapper = createQueryWrapper(dto);
        final List<ParkSpaceDate> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<ParkSpaceDateDTO> getListDto(ParkSpaceDateDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<ParkSpaceDateDTO> page(Page<ParkSpaceDate> page, ParkSpaceDateDTO dto) {
        final QueryWrapper<ParkSpaceDate> queryWrapper = createQueryWrapper(dto, "t1.");
        final IPage<ParkSpaceDateDTO> iPage = baseMapper.page(page, queryWrapper);
        Optional.ofNullable(iPage.getRecords()).ifPresent(l-> l.forEach(this::buildDisplay));
        queryWrapper.clear();
        return iPage;
    }

    /**
     * 根据唯一性索引获得数据
     * param tenant 
     * param leaseTime 
     *
     * @return 单条数据
     */
    public ParkSpaceDate getUniqueByTenantAndLeaseTime(String tenant, LocalDateTime leaseTime, String sid) {
        final QueryWrapper<ParkSpaceDate> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("sid", sid);
        queryWrapper.eq("tenant", tenant);
        queryWrapper.eq("lease_time", leaseTime);
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_PANDA_PARK, tenant, leaseTime, sid);
        final ParkSpaceDate model = redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getOne(queryWrapper, false), ParkSpaceDate.class);
        queryWrapper.clear();
        return model;
    }

    /**
     * 验证唯一性索引
     *
     * @param dto 页面对象
     */
     public void checkUnique(ParkSpaceDateDTO dto) {
         ParkSpaceDate model = getUniqueByTenantAndLeaseTime(dto.getTenant(), dto.getLeaseTime(), dto.getSid());
        if (null != model) {
            checkUnique(dto, model);
        }
    }

    /**
     * 验证唯一性索引
     *
     * @param dto   页面对象
     * @param model 根据约束字段查询出来的数据库对象
     */
    private void checkUnique(ParkSpaceDateDTO dto, ParkSpaceDate model) {
        if (!StringUtils.hasLength(dto.getDid())) {
            throw new ApiException("数据已存在，请勿重复操作");
        }
        if (!dto.getDid().equals(model.getDid())) {
            throw new ApiException("数据已存在，请勿重复操作");
        }
    }

    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(ParkSpaceDate model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getDid()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, model.getSid()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_PANDA_PARK, model.getTenant(), model.getLeaseTime(), model.getSid()));
    }

    /**
     * 批量清空缓存
     *
     * @param list 数据库对象
     */
    public void clearCaches(List<ParkSpaceDate> list) {
        if (null == list || list.isEmpty()) {
            return;
        }
        List<String> ids = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getDid())).collect(toList());
        if (!ids.isEmpty()) {
            redisTemplateUtil.deleteByKey(ids);
            ids.clear();
        }
        Set<String> uniqueIds = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_PANDA_PARK, model.getTenant(), model.getLeaseTime(), model.getSid())).collect(toSet());
        if (!uniqueIds.isEmpty()) {
            redisTemplateUtil.deleteByKey(uniqueIds);
            uniqueIds.clear();
        }
        Set<String> sidList = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, model.getSid())).collect(toSet());
        if (!sidList.isEmpty()) {
            try {
                log.info("clearCaches: {}", JacksonUtil.getObjectMapper().writeValueAsString(sidList));
            } catch (JsonProcessingException ignored) {
            }
            redisTemplateUtil.deleteByKey(sidList);
            sidList.clear();
        }
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public ParkSpaceDateDTO modelToDto(ParkSpaceDate model) {
        if (null == model) {
            return null;
        }
        final ParkSpaceDateDTO dto = new ParkSpaceDateDTO();
        dto.setDid(model.getDid());
        dto.setLeaseTime(model.getLeaseTime());
        dto.setSid(model.getSid());
        dto.setTenant(model.getTenant());
        dto.setTenantNumber(model.getTenantNumber());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public ParkSpaceDate dtoToModel(ParkSpaceDateDTO dto) {
        if (null == dto) {
            return null;
        }
        final ParkSpaceDate model = new ParkSpaceDate();
        model.setDid(dto.getDid());
        model.setLeaseTime(dto.getLeaseTime());
        model.setSid(dto.getSid());
        model.setTenant(dto.getTenant());
        model.setTenantNumber(dto.getTenantNumber());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(ParkSpaceDateDTO dto) {

    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<ParkSpaceDateDTO> modelToDtoList(List<ParkSpaceDate> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<ParkSpaceDate> dtoToModelList(List<ParkSpaceDateDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<ParkSpaceDate> createQueryWrapper(ParkSpaceDateDTO dto) {
        return createQueryWrapper(dto, "");
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto        查询参数，不为空的属性会被自动加入
     * @param tableAlias 表别名
     */
    public QueryWrapper<ParkSpaceDate> createQueryWrapper(ParkSpaceDateDTO dto, String tableAlias) {
        final QueryWrapper<ParkSpaceDate> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(tableAlias) && tableAlias.charAt(tableAlias.length() - 1) != '.') {
                tableAlias += ".";
            }
            if (StringUtils.hasLength(dto.getSid())) {
                queryWrapper.eq(tableAlias + "sid", dto.getSid());
            }
            if (StringUtils.hasLength(dto.getTenant())) {
                queryWrapper.eq(tableAlias + "tenant", dto.getTenant());
            }
        }
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     * @return 主键查询出来的集合
     */
    public List<ParkSpaceDate> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<ParkSpaceDate> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(ParkSpaceDate::getDid).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        dbIds.clear();
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }

    public List<ParkSpaceDate> getListBySid(String sid) {
        String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, sid);
        return redisTemplateUtil.getList(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> {
            QueryWrapper<ParkSpaceDate> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("sid", sid);
            queryWrapper.orderByAsc("lease_time");
            List<ParkSpaceDate> list = super.list(queryWrapper);
            queryWrapper.clear();
            return list;
        }, ParkSpaceDate.class);
    }


    @Transactional(rollbackFor = Exception.class)
    public void saveDateList(ParkSpaceDTO dto) {
        LocalDateTime startDate;
        LocalDateTime endDate;
        List<LocalDateTime> localDateTimeList;
        if (dto.getLeaseType().equals(PandaConstant.LEASE_TYPE_HOUR)) {
            startDate = DateUtil.strToLocalDateTime(dto.getStartDate() + " " + dto.getStartHour());
            endDate = DateUtil.strToLocalDateTime(dto.getStartDate() + " " + dto.getEndHour());
            localDateTimeList = DateUtil.getTimesBetween(startDate, endDate);
            if (dto.getEndHour().equals("23:59:59")) {
                localDateTimeList.add(LocalDateTime.of(startDate.getYear(), endDate.getMonthValue(), endDate.getDayOfMonth(), 23, 59, 59));
            }
        } else {
            startDate = DateUtil.strToLocalDateTime(dto.getStartDate() + " 00:00:00");
            endDate = DateUtil.strToLocalDateTime(dto.getEndDate() + " 00:00:00");
            localDateTimeList = DateUtil.getDatesBetween(startDate, endDate);
        }
        List<ParkSpaceDate> dateList = getListBySid(dto.getSid());
        if (dateList.isEmpty()) {
            for (LocalDateTime localDateTime : localDateTimeList) {
                saveDate(dto, localDateTime);
            }
        } else {
            List<ParkSpaceDate> deleteList = dateList.stream().filter(d -> !localDateTimeList.contains(d.getLeaseTime())).collect(toList());
            if (deleteList.stream().anyMatch(d -> StringUtils.hasLength(d.getTenant()))) {
                throw new ApiException("已经有人租赁,不能修改租赁时间");
            }
            if (!deleteList.isEmpty()) {
                List<String> ids = deleteList.stream().map(ParkSpaceDate::getDid).collect(toList());
                removeBatchIds(ids);
                ids.clear();
                deleteList.clear();
            }
            List<LocalDateTime> dbDateList = dateList.stream().map(ParkSpaceDate::getLeaseTime).collect(toList());
            for (LocalDateTime localDateTime : CollUtil.subtract(localDateTimeList, dbDateList)) {
                saveDate(dto, localDateTime);
            }
        }
    }

    public void saveDate(ParkSpaceDTO dto, LocalDateTime localDateTime) {
        ParkSpaceDateDTO parkSpaceDate = new ParkSpaceDateDTO();
        parkSpaceDate.setSid(dto.getSid());
        parkSpaceDate.setLeaseTime(localDateTime);
        save(parkSpaceDate);
        parkSpaceDate.freeData();
    }

    @Transactional(rollbackFor = Exception.class)
    public void removeBySid(String sid) {
        List<ParkSpaceDate> list = getListBySid(sid);
        if (!list.isEmpty()) {
            List<String> ids = list.stream().map(ParkSpaceDate::getDid).collect(toList());
            removeBatchIds(ids);
            ids.clear();
            list.clear();
        }
    }

    @Transactional(rollbackFor = Exception.class)
    public List<ParkSpaceDateDTO> getListBySidDTO(String sid) {
        return modelToDtoList(getListBySid(sid));
    }

    @Transactional(rollbackFor = Exception.class)
    public void updateParkSpaceDate(ParkSpaceTenantDTO dto) {
        if (!StringUtils.hasLength(dto.getTenant())) {
            throw new ApiException("租赁人不能为空");
        }
        if (!StringUtils.hasLength(dto.getTenantNumber())) {
            throw new ApiException("租赁人车牌号不能为空");
        }
        List<ParkSpaceDate> list = getListBySid(dto.getSid());
        if (!list.isEmpty()) {
            List<ParkSpaceDate> updateList = new ArrayList<>(list.size());
            List<LocalDateTime> dateList = dto.getDateList();
            for (ParkSpaceDate parkSpaceDate : list) {
                if (CollUtil.contains(dateList, parkSpaceDate.getLeaseTime())) {
                    if (StringUtils.hasLength(parkSpaceDate.getTenant())) {
                        if (!parkSpaceDate.getTenant().equals(dto.getTenant())) {
                            throw new ApiException("已经分配,不能操作");
                        }
                    }
                    parkSpaceDate.setTenant(dto.getTenant());
                    parkSpaceDate.setTenantNumber(dto.getTenantNumber());
                    updateList.add(parkSpaceDate);
                }
            }
            if (updateList.size() != dateList.size()) {
                log.error("数据异常,更新数量: {},前端传入数量: {}", updateList.size(), dateList.size());
                throw new ApiException("数据异常");
            }
            updateBatchIds(updateList);
            updateList.clear();
            list.clear();
            dateList.clear();
        }
    }

    @Transactional(readOnly = true)
    public boolean existsByTenantAndLeaseTime(LocalDateTime leaseTime, String sid) {
        QueryWrapper<ParkSpaceDate> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("sid", sid);
        queryWrapper.eq("lease_time", leaseTime);
        queryWrapper.isNotNull("tenant");
        boolean exists = baseMapper.exists(queryWrapper);
        queryWrapper.clear();
        return exists;
    }
}